#!/bin/bash
export ENABLE_INIT_ON_CPU=1
export TF_XLA_FLAGS="--tf_xla_auto_jit=-1 --tf_xla_min_cluster_size=4"
export DTU_UMD_FLAGS='ib_pool_size=134217728'
export ENFLAME_AUTO4C=true